package cc.sunni;

import cn.hutool.core.util.StrUtil;
import cn.hutool.extra.qrcode.QrCodeUtil;
import cn.hutool.extra.qrcode.QrConfig;
import sun.misc.BASE64Encoder;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;

/**
 * @author jiangli
 * @since 2021/1/6 16:37
 * 合成二维码工具类
 */
public class QrCoder {

    /**
     * 图片的宽度
     */
    private static final int IMAGE_WIDTH = 150;
    /**
     * 图片的高度(需按实际内容高度进行调整)
     */
    private static final int IMAGE_HEIGHT = 300;
    /**
     * 二维码的宽度
     */
    private static final int QR_CODE_WIDTH = 150;
    /**
     * 二维码的高度
     */
    private static final int QR_CODE_HEIGHT = 150;

    private static final String FORMAT_NAME = "JPG";

    /**
     * @param imgLogo logo图片
     * @param title   头部标题
     * @param content 内容
     * @param footer  底部文字
     */
    public static BufferedImage createQrCode(String imgLogo, String title, String content, String footer) {
        // 头部文字区域高度
        int titleHeight = 50;

        // 创建主模板图片
        BufferedImage image = new BufferedImage(IMAGE_WIDTH, IMAGE_HEIGHT, BufferedImage.TYPE_INT_RGB);
        Graphics2D main = image.createGraphics();
        // 设置图片的背景色
        main.setColor(Color.white); //白色
        main.fillRect(0, 0, IMAGE_WIDTH, IMAGE_HEIGHT);

        // 动态高度
        int height = 0;

        //***********************页面头部 文字****************
        Graphics2D titleRight = image.createGraphics();
        // 设置字体颜色 black黑 white白
        titleRight.setColor(Color.black);
        // 设置字体
        Font titleFont = new Font("宋体", Font.BOLD, 18);
        titleRight.setFont(titleFont);
        titleRight.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_LCD_HRGB);
        // 居中 x开始的位置：（图片宽度-字体大小*字的个数）/2
        int x = (IMAGE_WIDTH - (titleFont.getSize() * title.length())) / 2;
        titleRight.drawString(title, x, (titleHeight) / 2 + 10);
        height += titleHeight;

        //**********************中间文字部分*********
        Graphics2D centerWord = image.createGraphics();
        // 设置字体颜色，先设置颜色，再填充内容
        centerWord.setColor(Color.black);
        // 设置字体
        Font wordFont = new Font("宋体", Font.PLAIN, 10);
        centerWord.setFont(wordFont);
        centerWord.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_LCD_HRGB);
//        String[] info = content.split("\\$");
//        for (String s : info) {
//            // x开始的位置：（图片宽度-字体大小*字的个数）/2
//            int strWidth = centerWord.getFontMetrics().stringWidth(s);
//            // 总长度减去文字长度的一半  （居中显示）
//            int startX = (IMAGE_WIDTH - strWidth) / 2;
//            height += 20;
//            centerWord.drawString(s, startX, height);
//        }
        // 获取字符串字符的总宽度
        int stringLength = getStringLength(centerWord, content);
        // 获取字符高度
        int strHeight = getStringHeight(centerWord);
        // 绘制内容文字
        if (stringLength > IMAGE_WIDTH) {
            int rowstrnum = getRowStrNum(content.length(), IMAGE_WIDTH, stringLength);
            int rows = getRows(stringLength, IMAGE_WIDTH);
            String temp;
            for (int i = 0; i < rows; i++) {
                //获取各行的String
                if (i == rows - 1) {
                    //最后一行
                    temp = content.substring(i * rowstrnum);
                } else {
                    temp = content.substring(i * rowstrnum, i * rowstrnum + rowstrnum);
                }
                if (i > 0) {
                    //第一行不需要增加字符高度，以后的每一行在换行的时候都需要增加字符高度
                    height = height + strHeight;
                }
                centerWord.drawString(temp, 5, height);
            }
        } else {
            //直接绘制
            centerWord.drawString(content, 5, height);
        }
        //***************插入二维码图片***********************************************
        Graphics codePic = image.getGraphics();
        BufferedImage codeImg;
        QrConfig config = new QrConfig();
        config.setMargin(0);
        config.setWidth(QR_CODE_WIDTH);
        config.setHeight(QR_CODE_HEIGHT);
        if (StrUtil.isNotBlank(imgLogo)) {
            config.setImg(imgLogo);
        }
        codeImg = QrCodeUtil.generate(content, config);
        // 绘制二维码
        codePic.drawImage(codeImg, (IMAGE_WIDTH - QR_CODE_WIDTH) / 2, height, QR_CODE_WIDTH, QR_CODE_HEIGHT, null);
        codePic.dispose();

        //**********************底部公司名字*********
        Graphics2D typeLeft = image.createGraphics();
        // 设置字体颜色
        typeLeft.setColor(Color.black);
        // 设置字体
        Font footerFont = new Font("宋体", Font.PLAIN, 10);
        typeLeft.setFont(footerFont);
        typeLeft.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_LCD_HRGB);
        // x开始的位置：（图片宽度-字体大小*字的个数）/2
        int startX = (IMAGE_WIDTH - (footerFont.getSize() * footer.length())) / 2;
        // 获取字符高度
        int footerStrHeight = getStringHeight(typeLeft);
        height += QR_CODE_HEIGHT + footerStrHeight;
        typeLeft.drawString(footer, startX, height);

        //***************插入标志图片***********************************************
//        Graphics signPic = image.getGraphics();
//        BufferedImage signImg = null;
//        try {
//            signImg = ImageIO.read(new java.io.File(imgSign));
//        } catch (Exception e) {
//        }
//
//        if (signImg != null) {
//            signPic.drawImage(signImg, 0, 130, QR_CODE_WIDTH, QR_CODE_HEIGHT, null);
//            signPic.dispose();
//        }
        // 返回image方便后续处理是生成图片还是生成base64字符串
        return image;
    }

    // 生成图片文件
    public static void createImage(BufferedImage image, String fileLocation) {
        if (image != null) {
            try {
                ImageIO.write(image, "png", new File(fileLocation));
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    // 获取图片base64数据
    public static String base64ImageString(BufferedImage image) throws IOException {
        ByteArrayOutputStream bos = new ByteArrayOutputStream();//io流
        ImageIO.write(image, FORMAT_NAME, bos);//写入流中
        byte[] bytes = bos.toByteArray();//转换成字节
        BASE64Encoder encoder = new BASE64Encoder();
        String jpgBase64 = encoder.encodeBuffer(bytes).trim();//转换成base64串
        jpgBase64 = jpgBase64.replaceAll("\n", "").replaceAll("\r", "");//删除 \r\n
        return "data:image/jpg;base64," + jpgBase64;
    }

    // 字符串总宽度
    private static int getStringLength(Graphics g, String str) {
        char[] chars = str.toCharArray();
        return g.getFontMetrics().charsWidth(chars, 0, str.length());
    }

    // 字符高度
    private static int getStringHeight(Graphics g) {
        return g.getFontMetrics().getHeight();
    }

    // 每一行字符的个数
    private static int getRowStrNum(int strNum, int rowWidth, int strWidth) {
        return (rowWidth * strNum) / strWidth;
    }

    // 字符行数
    private static int getRows(int strWidth, int rowWidth) {
        int rows;
        if (strWidth % rowWidth > 0) {
            rows = strWidth / rowWidth + 1;
        } else {
            rows = strWidth / rowWidth;
        }
        return rows;
    }

    public static void main(String[] args) {
        String content = "汇丰管业-HDPE节流式缠绕增强管-DN300-20210101-A001-0821001-080901000001";
        BufferedImage bufferedImage = createQrCode(null, "物资模块", content, "三峡科技有限公司");
        createImage(bufferedImage, "D:/demo.test.png");
    }
}
